import numpy as np

print(np.ones((2, 3)))
print(np.zeros((4, 3)))
t = np.eye(10)
print(t)

'''
在numpy中可以理解为方向,使用0,1,2...数字表示,对于一个一维数组,只有一个0轴,
对于2维数组(shape(2,2)),有0轴和1轴,对于三维数组(shape(2,2,3)),有0,1,2轴
'''
t = np.eye(4)
print("t=", t)
print("np.argmax(t, axis=0)=", np.argmax(t, axis=0)) # 计算行方向上的值  TODO 可能注释有错
print("np.argmax(t, axis=1)=", np.argmax(t, axis=1)) # 计算列方向上的值

t[t == 1] = -1
print("t=", t)
print("np.argmin(t, axis=1)=", np.argmin(t, axis=1))
print("np.argmin(t, axis=0)=", np.argmin(t, axis=0))


print(np.random.randint(10,20,(4, 5)))